import os
from utils.report_utils import get_report_path

if __name__ == "__main__":
    # 调用函数获取报告路径
    report_path = get_report_path()
    # 构建 Locust 命令
    # GUI模式下的命令
    command = f"locust -f locustfile.py --html={report_path}"
    # 非GUI模式下的命令
    # command = f"locust -f locustfile.py --headless -u 10 -r 1 --html={report_path} -t 10s"
    try:
        # 执行命令
        os.system(command)
        print(f"Locust 压测已执行，报告保存路径为: {report_path}")
    except Exception as e:
        print(f"执行 Locust 压测时出错: {e}")
